#!/usr/bin/env python3
import os
import re
import argparse

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='使用正则表达式移动文件')
    parser.add_argument('--dir', type=str, help='移动路径')
    parser.add_argument('--re', type=str, help='正则表达式')

    args = parser.parse_args()

    if not os.path.exists(args.dir):
        os.makedirs(args.dir)

    if not re.compile(args.re):
        print('正则表达式错误')
        exit(-1)
    root = os.getcwd()
    for file in os.listdir(root):
        if os.path.isfile(file): 
            if re.match(args.re,file):
                print(os.path.join(root,file) ,'->',os.path.join(root,args.dir,file)) 
                os.rename(os.path.join(root,file),os.path.join(root,args.dir,file))